"""
提示词工程模块 - 管理与ERNIE模型交互的提示词
"""

def create_system_prompt(artifact_name=None):
    """创建系统提示词"""
    prompt = """你是一位专业的动物讲解员和动物学专家，擅长深入浅出地解释动物习性。
    
    请注意以下几点：
    1. 你的回答应既有学术价值又生动有趣，适合各年龄层观众理解
    2. 尽量使用生动形象的描述和比喻，帮助用户理解复杂的历史背景和艺术特点
    3. 回答应条理清晰，重点突出，可适当分段
    4. 引用具体细节和事实，避免泛泛而谈
    5. 如果知识库没有信息，或不确定的内容，请坦率承认
    6. 保持友好亲切的语气，像正在进行面对面讲解一样

    在回答时，可考虑以下角度：
    - 动物的食性和栖息地特征
    - 动物的天敌和食物
    - 动物的演变历史
    """
    
    if artifact_name:
        prompt += f"\n\n现在，用户正在了解《{artifact_name}》，请特别关注这个动物的相关信息。"
    
    return prompt

def create_context_prompt(query, context):
    """创建上下文提示词"""
    return f"""参考以下知识回答问题。如果知识库中没有相关信息，请基于你的专业知识回答，并明确指出这是你的补充解释。

知识库内容:
{context}

用户询问: {query}"""

def generate_quick_queries(artifact_name):
    """生成与文物相关的快速查询按钮"""
    standard_queries = [
        f"{artifact_name}的生活习性是什么？",
        f"{artifact_name}的栖息地在哪里？",
        f"{artifact_name}的天敌和食物是什么？"
    ]
    """
    custom_queries = {
        "青铜器": [f"青铜器铸造工艺是怎样的？", f"不同时期青铜器的区别是什么？"],
        "兵马俑": [f"兵马俑的发现过程是怎样的？", f"兵马俑反映了秦朝哪些军事特点？"],
        "敦煌壁画": [f"敦煌壁画的色彩和绘画技法是什么？", f"敦煌壁画中的佛教故事有哪些？"],
        "唐三彩": [f"唐三彩的烧制工艺是怎样的？", f"唐三彩在丝绸之路贸易中的作用是什么？"],
        "越王勾践剑": [f"越王勾践剑的锻造技术有何特点？", f"越王勾践的历史故事是什么？"],
        "甲骨文": [f"甲骨文的发现过程是怎样的？", f"甲骨文对研究商代历史有何价值？"],
        "四羊方尊": [f"四羊方尊的象征意义是什么？", f"四羊方尊上的纹饰代表什么？"],
        "汉长信宫灯": [f"汉长信宫灯的设计有何创新？", f"汉长信宫灯反映了汉代什么技术？"],
        "曾侯乙编钟": [f"曾侯乙编钟的音律特点是什么？", f"编钟在古代礼乐文化中的地位如何？"],
        "清明上河图": [f"清明上河图描绘了哪些宋代社会场景？", f"清明上河图的艺术技法有何特点？"]
    }
    """

    custom_queries = {
        "狮子": [f"狮子的主要栖息地有哪些特点？", f"狮子的食物构成及捕食方式是怎样的？"],
        "老虎": [f"老虎的典型栖息地环境有何特征？", f"老虎的主要食物来源及觅食习惯是什么？"],
        "大象": [f"大象适宜的栖息地有哪些生态特点？", f"大象的食物种类及每日食量是多少？"],
        "长颈鹿": [f"长颈鹿的栖息地通常具备哪些条件？", f"长颈鹿的食物偏好及取食方式有何特点？"],
        "斑马": [f"斑马主要栖息在什么样的环境中？", f"斑马的食物构成及进食习惯是怎样的？"],
        "企鹅": [f"企鹅的栖息地有哪些独特的地理和气候特征？", f"企鹅的食物来源及捕食策略是什么？"],
        "北极熊": [f"北极熊的栖息地环境随季节有何变化？", f"北极熊的主要食物及觅食依赖的环境条件是什么？"],
        "海豚": [f"海豚的栖息地通常分布在哪些海域？", f"海豚的食物种类及捕食方式有何特点？"],
        "孔雀": [f"孔雀的自然栖息地有哪些生态特点？", f"孔雀的食物构成及进食习性是怎样的？"],
        "金丝猴": [f"金丝猴的栖息地主要分布在哪些区域？有何环境特征？", f"金丝猴的食物种类及取食习惯是什么？"]
    }
    all_queries = standard_queries.copy()
    if artifact_name in custom_queries:
        all_queries.extend(custom_queries[artifact_name])
    
    return all_queries 